package com.yzg.simple.nio;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Arrays;

/**
 * @author misterWei
 * @create 2019年12月21号:22点23分
 * @mailbox mynameisweiyan@gmail.com
 *
 * 对Buffer 的 分散 聚合 多维度操作
 */
public class ScatteringAndGatheringTest {


    public static void main(String[] args) throws Exception {
        //使用 ServerSocketChannel 和 SocketChannel 网络
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        InetSocketAddress inetSocketAddress = new InetSocketAddress(7000); //绑定端口到 socket ，并启动
        serverSocketChannel.socket().bind(inetSocketAddress);
        //创建 buffer 数组
        ByteBuffer[] byteBuffers = new ByteBuffer[2];
        byteBuffers[0] = ByteBuffer.allocate(5);
        byteBuffers[1] = ByteBuffer.allocate(3);
        //等客户端连接(telnet)
        SocketChannel socketChannel = serverSocketChannel.accept();
        int messageLength = 8;
        //假定从客户端接收 8 个字节
        // 循环的读取

        /**
         * 这段代码的意思详解
         *
         *  定义一个死循环,去监听  里面有 2个 条件循环
         *  第一个条件循环去读取阻塞,第二个条件循环是把当前Buffer写出
         *
         *  字节条件为 8  临界点
         *  如果当前字节 小于8,那么就一直会在第一个条件中去读取
         *  如果当前字节 大与8 , 那么Buffer 就会不够用,如果不够用,就会走入最后一步就是clear操作,然后读取就有位置了.
         *
         *  注意,如果不clear 那么循环读取,循环Buffer永远都是满的,那么就会陷入死循环
         *
         *  总结一句话: read Buffer 如果Buffer 不够用,那么就会先去读取,然后循环再去read 直到读取结束.

         */
        while (true) {
            int byteRead = 0;
            while (byteRead < messageLength) {
                // 如果消息读不到就开始阻塞
                long l = socketChannel.read(byteBuffers);
                byteRead += l;
                //累计读取的字节数
                System.out.println("byteRead=" + byteRead);
                //使用流打印, 看看当前的这个 buffer 的 position 和 limit
                Arrays.asList(byteBuffers).stream().map(buffer -> "postion=" + buffer.position() + ", limit=" + buffer.limit()).forEach(System.out::println);
            }

            //将所有的 buffer 进行 flip
            Arrays.asList(byteBuffers).forEach(buffer -> buffer.flip());

            //将数据读出显示到客户端
            long byteWirte = 0;
            while (byteWirte < messageLength) {
                long l = socketChannel.write(byteBuffers);
                byteWirte += l;
            }

            //将所有的 buffer 进行 clear
            /*Arrays.asList(byteBuffers).forEach(buffer -> {
                buffer.clear();
            });*/
            System.out.println("byteRead:=" + byteRead + " byteWrite=" + byteWirte + ", messagelength" + messageLength);
        }

    }

}
